package com.hongshu.file.controller;

import com.hongshu.common.core.domain.R;
import com.hongshu.common.core.utils.file.FileUtils;
import com.hongshu.file.service.ISysFileService;
import com.hongshu.system.api.domain.SysFile;
import org.apache.commons.lang3.ArrayUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.concurrent.CompletableFuture;

/**
 * 文件请求处理
 *
 * @author hongshu
 */
@RestController
public class SysFileController {

    private static final Logger log = LoggerFactory.getLogger(SysFileController.class);

    @Autowired
    private ISysFileService sysFileService;


    private static final long MAX_TOTAL_SIZE = 50 * 1024 * 1024; // 总大小限制50MB
    private static final int MAX_FILES = 10; // 最大文件数量


    /**
     * 文件上传请求
     */
    @PostMapping("upload")
    public R<SysFile> upload(MultipartFile file) {
        // 参数校验
        if (file == null || file.isEmpty()) {
            return R.fail("上传文件不能为空");
        }
        try {
            // 上传并返回访问地址
            String url = sysFileService.uploadFile(file);
            SysFile sysFile = new SysFile();
            sysFile.setName(FileUtils.getName(url));
            sysFile.setUrl(url);
            return R.ok(sysFile);
        } catch (Exception e) {
            log.error("上传文件失败", e);
            return R.fail(e.getMessage());
        }
    }

    @PostMapping("batch/upload")
    public R<List<SysFile>> batchUpload(MultipartFile[] files) {
        // 参数校验
        if (ArrayUtils.isEmpty(files)) {
            return R.fail("上传文件不能为空");
        }
        // 文件数量校验
        if (files.length > MAX_FILES) {
            return R.fail("单次最多上传" + MAX_FILES + "个文件");
        }
        try {
            // 总大小校验
//            long totalSize = Arrays.stream(files)
//                    .mapToLong(MultipartFile::getSize)
//                    .sum();
//            if (totalSize > MAX_TOTAL_SIZE) {
//                return R.fail("文件总大小不能超过50MB");
//            }
            List<SysFile> resultList = new ArrayList<>();
            List<String> failedFiles = new ArrayList<>();
            // 并行上传文件
            CompletableFuture<?>[] futures = Arrays.stream(files)
                    .map(file -> CompletableFuture.runAsync(() -> {
                        try {
                            // 单个文件上传
                            String url = sysFileService.uploadFile(file);
                            SysFile sysFile = SysFile.builder()
                                    .name(FileUtils.getName(url))
                                    .url(url)
//                                    .size(file.getSize())
//                                    .contentType(file.getContentType())
//                                    .originalFilename(file.getOriginalFilename())
//                                    .createTime(LocalDateTime.now())
                                    .build();
                            synchronized (resultList) {
                                resultList.add(sysFile);
                            }
                        } catch (Exception e) {
                            log.error("文件上传失败: {}", file.getOriginalFilename(), e);
                            synchronized (failedFiles) {
                                failedFiles.add(file.getOriginalFilename());
                            }
                        }
                    }))
                    .toArray(CompletableFuture[]::new);
            // 等待所有上传完成
            CompletableFuture.allOf(futures).join();
            // 处理上传结果
            if (!failedFiles.isEmpty()) {
                return R.fail("部分文件上传失败: " + String.join(", ", failedFiles));
            }
            return R.ok(resultList);
        } catch (Exception e) {
            log.error("批量上传文件失败", e);
            return R.fail("批量上传失败：" + e.getMessage());
        }
    }


    /**
     * 文件上传请求
     */
    @PostMapping("upload/file")
    public R<SysFile> upload(File file) {
        // 参数校验
        if (file == null) {
            return R.fail("上传文件不能为空");
        }
        try {
            // 上传并返回访问地址
            String url = sysFileService.uploadFile(file);
            SysFile sysFile = new SysFile();
            sysFile.setName(FileUtils.getName(url));
            sysFile.setUrl(url);
            return R.ok(sysFile);
        } catch (Exception e) {
            log.error("上传文件失败", e);
            return R.fail(e.getMessage());
        }
    }

    @PostMapping("batch/upload/file")
    public R<List<SysFile>> batchUpload(File[] files) {
        // 参数校验
        if (ArrayUtils.isEmpty(files)) {
            return R.fail("上传文件不能为空");
        }
        // 文件数量校验
        if (files.length > MAX_FILES) {
            return R.fail("单次最多上传" + MAX_FILES + "个文件");
        }
        try {
            // 总大小校验
//            long totalSize = Arrays.stream(files)
//                    .mapToLong(MultipartFile::getSize)
//                    .sum();
//            if (totalSize > MAX_TOTAL_SIZE) {
//                return R.fail("文件总大小不能超过50MB");
//            }
            List<SysFile> resultList = new ArrayList<>();
            List<String> failedFiles = new ArrayList<>();
            // 并行上传文件
            CompletableFuture<?>[] futures = Arrays.stream(files)
                    .map(file -> CompletableFuture.runAsync(() -> {
                        try {
                            // 单个文件上传
                            String url = sysFileService.uploadFile(file);
                            SysFile sysFile = SysFile.builder()
                                    .name(FileUtils.getName(url))
                                    .url(url)
//                                    .size(file.getSize())
//                                    .contentType(file.getContentType())
//                                    .originalFilename(file.getOriginalFilename())
//                                    .createTime(LocalDateTime.now())
                                    .build();
                            synchronized (resultList) {
                                resultList.add(sysFile);
                            }
                        } catch (Exception e) {
//                            log.error("文件上传失败: {}", file.getOriginalFilename(), e);
                            synchronized (failedFiles) {
//                                failedFiles.add(file.getOriginalFilename());
                            }
                        }
                    }))
                    .toArray(CompletableFuture[]::new);
            // 等待所有上传完成
            CompletableFuture.allOf(futures).join();
            // 处理上传结果
            if (!failedFiles.isEmpty()) {
                return R.fail("部分文件上传失败: " + String.join(", ", failedFiles));
            }
            return R.ok(resultList);
        } catch (Exception e) {
            log.error("批量上传文件失败", e);
            return R.fail("批量上传失败：" + e.getMessage());
        }
    }
}
